package tree;

/**
 * @author: june
 * @date: 2023/7/13
 * <p>
 * 617. 合并二叉树
 * 给你两棵二叉树： root1 和 root2 。
 * 想象一下，当你将其中一棵覆盖到另一棵之上时，两棵树上的一些节点将会重叠（而另一些不会）。你需要将这两棵树合并成一棵新二叉树。
 * 合并的规则是：如果两个节点重叠，那么将这两个节点的值相加作为合并后节点的新值；
 * 否则，不为 null 的节点将直接作为新二叉树的节点。
 * <p>
 * 返回合并后的二叉树。
 * 注意: 合并过程必须从两个树的根节点开始。
 **/

public class Merge2Tree_617 {
    public TreeNode mergeTrees(TreeNode root1, TreeNode root2) {
        if (root1 == null) {
            return root2;
        }

        if (root2 == null) {
            return root1;
        }

        TreeNode root = new TreeNode(root1.val + root2.val);
        if (root1.left == null && root2.left != null) {
            root.left = root2.left;
        } else if (root1.left != null && root2.left == null) {
            root.left = root1.left;
        } else {
            root.left = mergeTrees(root1.left, root2.left);
        }

        if (root1.right == null && root2.right != null) {
            root.right = root2.right;
        } else if (root1.right != null && root2.right == null) {
            root.right = root1.right;
        } else {
            root.right = mergeTrees(root1.right, root2.right);
        }
        return root;
    }

    // 对第一种方法的简化版
    public TreeNode mergeTrees2(TreeNode root1, TreeNode root2) {
        if (root1 == null) {
            return root2;
        }

        if (root2 == null) {
            return root1;
        }

        TreeNode root = new TreeNode(root1.val + root2.val);
        root.left = mergeTrees2(root1.left, root2.left);
        root.right = mergeTrees2(root1.right, root2.right);
        return root;
    }
}